money = 5000000
name = input("What is your name? ")


# 定义查询函数
def query(show_header):
    if show_header:
        print("---------查询余额----------")
    print(f"{name} hello,你还剩{money}元")


# 定义存款函数
def saving(num):
    global money     # money在函数内部定义为全局变量
    money += num
    print("---------存入金额---------")
    print(f"{name} hello,存入{num}元成功")
    query(False)     # 查询余额


# 定义取款函数
def get_money(num):
    global money
    money -= num
    print("---------  取款  ---------")
    print(f"{name}，你好，你取款{num}元成功")
    query(False)    # 查询余额


# 定义主菜单函数
def main():
    print("--------- 主菜单 ----------")
    print(f"{name},你好，欢迎使用ATM，请选择操作：")
    print("[1]\t查询余额")     # 通过\t对齐输出
    print("[2]\t存款")
    print("[3]\t取款")
    print("[4]\t退出")
    return input("请输入你的选择：")


# 设置无限循环，确保程序不退出
while True:
    keyboard_input = main()
    if keyboard_input == "1":
        query(True)
        continue     # 通过continue继续下一次循环，一进来就回到了主菜单
    elif keyboard_input == "2":
        num = int(input("你想要存多少钱？请输入："))
        saving(num)
        continue
    elif keyboard_input == "3":
        num = int(input("你想要取出多少钱？请输入："))
        get_money(num)
        continue
    else:
        print("程序退出")
        break








